import os
import torch
import json

class Config(object):
    def __init__(self):
        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
        self.train_file_path = r"D:\AI_workplace\GitProjects\mystudy_knowledge_graph\data\train.txt"
        self.vocab_path = r"D:\AI_workplace\GitProjects\mystudy_knowledge_graph\vocab\vocab.txt"
        self.tag2id = json.load(open(r"D:\AI_workplace\GitProjects\mystudy_knowledge_graph\data\tag2id.json", 'r', encoding='utf-8'))
        self.embedding_dim = 512
        self.epochs = 5
        self.batch_size = 16
        self.hidden_dim = 256
        self.lr = 1e-3
        self.dropout = 0.2
        self.model = "BiLSTM_CRF"



if __name__ == '__main__':
    config = Config()
    print(config.device)
    print(config.train_file_path)
    print(config.vocab_path)
    print(config.tag2id)